import torch

# device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
# print(f"Using device: {device}")

print(torch.__version__)           # 查看 PyTorch 版本
print(torch.cuda.is_available())   # 返回 True 表示 GPU 可用
print(torch.cuda.get_device_name(0))  # 显示显卡型号

